from __future__ import absolute_import,division,print_function
import pandas as pd
from sklearn.model_selection import train_test_split
import os
# dirname="data"
# os.mkdir("./"+dirname)
data=pd.read_csv("./diabetes.csv")

train,test=train_test_split(data,test_size=0.4)
test,val=train_test_split(test,test_size=0.5,random_state=42)


print("训练数据",len(train))
print("验证数据",len(val))
print("测试数据",len(test))


# train.to_csv("./data/train.csv",index=False)
# val.to_csv("./data/val.csv",index=False)
# test.to_csv("./data/test.csv",index=False)


